/**
 * @file lv_port_disp.c
 *
 */

/*Copy this file as "lv_port_disp.c" and set this value to "1" to enable content*/
#if 1

/*********************
 *      INCLUDES
 *********************/
#include "lv_port_disp.h"
#include "st7789.h"
#include "esp_attr.h"
#include "esp_heap_caps.h"

/*********************
 *      DEFINES
 *********************/
#ifndef MY_DISP_HOR_RES
    #warning Please define or replace the macro MY_DISP_HOR_RES with the actual screen width, default value 320 is used for now.
    #define MY_DISP_HOR_RES    TFT_HEIGHT
#endif

#ifndef MY_DISP_VER_RES
    #warning Please define or replace the macro MY_DISP_HOR_RES with the actual screen height, default value 240 is used for now.
    #define MY_DISP_VER_RES    TFT_WIDTH
#endif

/**********************
 *      TYPEDEFS
 **********************/

/**********************
 *  STATIC PROTOTYPES
 **********************/
static void disp_init(void);

static void disp_flush(lv_disp_drv_t * disp_drv, const lv_area_t * area, lv_color_t * color_p);

//static void gpu_fill(lv_disp_drv_t * disp_drv, lv_color_t * dest_buf, lv_coord_t dest_width,
//        const lv_area_t * fill_area, lv_color_t color);

/**********************
 *  STATIC VARIABLES
 **********************/

/**********************
 *      MACROS
 **********************/

/**********************
 *   GLOBAL FUNCTIONS
 **********************/

void lv_port_disp_init(void)
{
    /*-------------------------
     * Initialize your display
     * -----------------------*/
    disp_init();

    /*-----------------------------
     * Create a buffer for drawing
     *----------------------------*/

    /**
     * LVGL requires a buffer where it internally draws the widgets.
     * Later this buffer will passed to your display driver's `flush_cb` to copy its content to your display.
     * The buffer has to be greater than 1 display row
     *
     * There are 3 buffering configurations:
     * 1. Create ONE buffer:
     *      LVGL will draw the display's content here and writes it to your display
     *
     * 2. Create TWO buffer:
     *      LVGL will draw the display's content to a buffer and writes it your display.
     *      You should use DMA to write the buffer's content to the display.
     *      It will enable LVGL to draw the next part of the screen to the other buffer while
     *      the data is being sent form the first buffer. It makes rendering and flushing parallel.
     *
     * 3. Double buffering
     *      Set 2 screens sized buffers and set disp_drv.full_refresh = 1.
     *      This way LVGL will always provide the whole rendered screen in `flush_cb`
     *      and you only need to change the frame buffer's address.
     */

    /* Example for 1) */
    // static lv_disp_draw_buf_t draw_buf_dsc_1;
    // static lv_color_t buf_1[MY_DISP_HOR_RES * 40];                          /*A buffer for 10 rows*/
    // lv_disp_draw_buf_init(&draw_buf_dsc_1, buf_1, NULL, MY_DISP_HOR_RES * 40);   /*Initialize the display buffer*/

    /* Example for 2) */
    static lv_disp_draw_buf_t draw_buf_dsc_2;
    // static lv_color_t buf_2_1[MY_DISP_HOR_RES * 80];                        /*A buffer for 10 rows*/
    // static lv_color_t buf_2_2[MY_DISP_HOR_RES * 80];                        /*An other buffer for 10 rows*/

    lv_color_t *buf1 = heap_caps_malloc(MY_DISP_HOR_RES * 45 * sizeof(lv_color_t), MALLOC_CAP_DMA);
    lv_color_t *buf2 = heap_caps_malloc(MY_DISP_HOR_RES * 45 * sizeof(lv_color_t), MALLOC_CAP_DMA);

    lv_disp_draw_buf_init(&draw_buf_dsc_2, buf1, buf2, MY_DISP_HOR_RES * 45);   /*Initialize the display buffer*/

    /* Example for 3) also set disp_drv.full_refresh = 1 below*/
    // lv_disp_draw_buf_t draw_buf_dsc_3;
    // lv_color_t buf_3_1[MY_DISP_HOR_RES * MY_DISP_VER_RES];            /*A screen sized buffer*/
    // lv_color_t buf_3_2[MY_DISP_HOR_RES * MY_DISP_VER_RES];            /*Another screen sized buffer*/
    // lv_disp_draw_buf_init(&draw_buf_dsc_3, buf_3_1, NULL,
    //                       MY_DISP_VER_RES * MY_DISP_HOR_RES);   /*Initialize the display buffer*/

    /*-----------------------------------
     * Register the display in LVGL
     *----------------------------------*/

    static lv_disp_drv_t disp_drv;                         /*Descriptor of a display driver*/
    lv_disp_drv_init(&disp_drv);                    /*Basic initialization*/

    /*Set up the functions to access to your display*/
    // disp_drv.rounder_cb = disp_rounder_cb;
    // disp_drv.set_px_cb = disp_set_px_cb;

    /*Set the resolution of the display*/
    disp_drv.hor_res = MY_DISP_HOR_RES;
    disp_drv.ver_res = MY_DISP_VER_RES;

    /*Used to copy the buffer's content to the display*/
    disp_drv.flush_cb = disp_flush;

    /*Set a display buffer*/
    disp_drv.draw_buf = &draw_buf_dsc_2;

    /*Required for Example 3)*/
    // disp_drv.full_refresh = 1;

    /* Fill a memory array with a color if you have GPU.
     * Note that, in lv_conf.h you can enable GPUs that has built-in support in LVGL.
     * But if you have a different GPU you can use with this callback.*/
    //disp_drv.gpu_fill_cb = gpu_fill;

    /*Finally register the driver*/
    lv_disp_drv_register(&disp_drv);
}

/**********************
 *   STATIC FUNCTIONS
 **********************/

/*Initialize your display and the required peripherals.*/
static void disp_init(void)
{
    /*You code here*/
    st7789_init();
}

volatile bool disp_flush_enabled = true;

/* Enable updating the screen (the flushing process) when disp_flush() is called by LVGL
 */
void disp_enable_update(void)
{
    disp_flush_enabled = true;
}

/* Disable updating the screen (the flushing process) when disp_flush() is called by LVGL
 */
void disp_disable_update(void)
{
    disp_flush_enabled = false;
}


/*Flush the content of the internal buffer the specific area on the display
 *You can use DMA or any hardware acceleration to do this operation in the background but
 *'lv_disp_flush_ready()' has to be called when finished.*/
/* The ST7789 display controller can drive up to 320*240 displays, when using a 240*240 or 240*135
 * displays there's a gap of 80px or 40/52/53px respectively. 52px or 53x offset depends on display orientation.
 * We need to edit the coordinates to take into account those gaps, this is not necessary in all orientations. */
static void disp_flush(lv_disp_drv_t * disp_drv, const lv_area_t * area, lv_color_t * color_p)
{
    if(disp_flush_enabled) {
        uint8_t data[4] = {0};

            uint16_t offsetx1 = area->x1;
            uint16_t offsetx2 = area->x2;
            uint16_t offsety1 = area->y1;
            uint16_t offsety2 = area->y2;

        #if (CONFIG_LV_TFT_DISPLAY_OFFSETS)
            offsetx1 += CONFIG_LV_TFT_DISPLAY_X_OFFSET;
            offsetx2 += CONFIG_LV_TFT_DISPLAY_X_OFFSET;
            offsety1 += CONFIG_LV_TFT_DISPLAY_Y_OFFSET;
            offsety2 += CONFIG_LV_TFT_DISPLAY_Y_OFFSET;

        #elif (MY_DISP_HOR_RES == 240) && (MY_DISP_VER_RES == 240)
            #if (CONFIG_LV_DISPLAY_ORIENTATION_PORTRAIT)
                offsetx1 += 80;
                offsetx2 += 80;
            #elif (CONFIG_LV_DISPLAY_ORIENTATION_LANDSCAPE_INVERTED)
                offsety1 += 80;
                offsety2 += 80;
            #endif
        #elif (MY_DISP_HOR_RES == 240) && (MY_DISP_VER_RES == 135)
            #if (CONFIG_LV_DISPLAY_ORIENTATION_PORTRAIT) || \
                (CONFIG_LV_DISPLAY_ORIENTATION_PORTRAIT_INVERTED)
                offsetx1 += 40;
                offsetx2 += 40;
                offsety1 += 53;
                offsety2 += 53;
            #endif
        #elif (MY_DISP_HOR_RES == 135) && (MY_DISP_VER_RES == 240)
            #if (CONFIG_LV_DISPLAY_ORIENTATION_LANDSCAPE) || \
                (CONFIG_LV_DISPLAY_ORIENTATION_LANDSCAPE_INVERTED)
                offsetx1 += 52;
                offsetx2 += 52;
                offsety1 += 40;
                offsety2 += 40;
            #endif
        #endif

        /*Column addresses*/
        st7789_send_cmd(ST7789_CASET);
        data[0] = (offsetx1 >> 8) & 0xFF;
        data[1] = offsetx1 & 0xFF;
        data[2] = (offsetx2 >> 8) & 0xFF;
        data[3] = offsetx2 & 0xFF;
        st7789_send_data(data, 4);

        /*Page addresses*/
        st7789_send_cmd(ST7789_RASET);
        data[0] = (offsety1 >> 8) & 0xFF;
        data[1] = offsety1 & 0xFF;
        data[2] = (offsety2 >> 8) & 0xFF;
        data[3] = offsety2 & 0xFF;
        st7789_send_data(data, 4);

        /*Memory write*/
        st7789_send_cmd(ST7789_RAMWR);

        size_t size = (size_t)lv_area_get_width(area) * (size_t)lv_area_get_height(area);

        st7789_send_color((void*)color_p, size * 2);

    }

    /*IMPORTANT!!!
     *Inform the graphics library that you are ready with the flushing*/
    lv_disp_flush_ready(disp_drv);
}

/*OPTIONAL: GPU INTERFACE*/

/*If your MCU has hardware accelerator (GPU) then you can use it to fill a memory with a color*/
//static void gpu_fill(lv_disp_drv_t * disp_drv, lv_color_t * dest_buf, lv_coord_t dest_width,
//                    const lv_area_t * fill_area, lv_color_t color)
//{
//    /*It's an example code which should be done by your GPU*/
//    int32_t x, y;
//    dest_buf += dest_width * fill_area->y1; /*Go to the first line*/
//
//    for(y = fill_area->y1; y <= fill_area->y2; y++) {
//        for(x = fill_area->x1; x <= fill_area->x2; x++) {
//            dest_buf[x] = color;
//        }
//        dest_buf+=dest_width;    /*Go to the next line*/
//    }
//}


#else /*Enable this file at the top*/

/*This dummy typedef exists purely to silence -Wpedantic.*/
typedef int keep_pedantic_happy;
#endif
